import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class NaryTreeLevelOrderTraversal {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> ret = new ArrayList<>();
        if (root == null) {
            return ret;
        }

        Queue<Node> queue = new LinkedList<>();
        // 先将根节点入队
        queue.offer(root);

        while (!queue.isEmpty()) {
            // 记录本层的节点个数
            int size = queue.size();
            List<Integer> list = new ArrayList<>();
            // 因题目需要将每层节点分组, 因此 size 的作用就是给每层节点分组
            while (size != 0) {
                Node cur = queue.poll();
                list.add(cur.val);
                for (Node child : cur.children) {
                    // 遍历当前节点的孩子节点数组, 并添加
                    if (child != null) {
                        queue.offer(child);
                    }
                }
                size--;
            }
            ret.add(list);
        }
        return ret;
    }
}
